import csv
import time
import pymongo

# 使用Python自带的csv模块读取csv文件,并将其转换为包含字典的列表。其中每一个字典为csv中的一行数据。
with open('people_info.csv', encoding='utf-8') as f:
    reader = csv.DictReader(f)
    people_info_list = [x for x in reader]
# 初始化MongoDB,并连接到chapter_8库下面的batch集合。
handler = pymongo.MongoClient().chapter_8.batch

start_time = time.time()  # 记录开始时间戳。
handler.insert_many(people_info_list)  # 使用 insert_many()方法直接把包含字典的列表插入数据库。
end_time = time.time()  # 记录结束时间戳
print('批量插入数据，耗时：', end_time - start_time)  # 打印时间差
